python image drawing code